#include <iostream>
#include <memory>
#include "UdpServer.hpp"
#include "Route.hpp"
#include "ThreadPool.hpp"

using task_t =std::function<void()>;
using namespace ThreadPoolModule;

std::string defaulthandler(const std::string &message)
{
    std::string hello="hello";
    hello+=message;
    return hello;
}

//./udpserver ip port
int main(int argc,char *argv[])
{
    if(argc!=2)
    {
        std::cerr<<"Usage: "<<argv[0]<<"port"<<std::endl;
        return 1;
    }
    //std::string ip=argv[1];
    uint16_t port=std::stoi(argv[1]);
    Enable_Console_Log_Strategy();

    //1.路由服务
    Route r;

    //2.线程池
    auto tp=ThreadPool<task_t>::GetInstance();

    //3.网络服务器对象，提供通信功能
    std::unique_ptr<UdpServer> usvr = std::make_unique<UdpServer>(port, [&r, &tp](int sockfd, const std::string &message, InetAddr&peer){
        task_t t = std::bind(&Route::MessageRoute,&r, sockfd, message, peer);
        tp->Enqueue(t);
    });
    // std::unique_ptr<UdpServer> usvr=std::make_unique<UdpServer>(port,[&r](int sockfd,const std::string &message,InetAddr &peer)
    // {
    //     r.MessageRoute(sockfd,message,peer);
    // });
    usvr->Init();
    usvr->Start();
    return 0;
}